from wenet.zipformer.subsampling import Conv2dSubsampling
from wenet.zipformer.scaling import ScheduledFloat


def init_encoder_embedding():
    encoder_embed = Conv2dSubsampling(
        in_channels=80,
        out_channels=192,
        dropout=ScheduledFloat((0.0, 0.3), (20000.0, 0.1)),
    )
    return encoder_embed